powerful covid-19 mask detection tool
Creating a Powerful COVID-19 Mask Detection Tool with PyTorch
Using the MaskedFace-Net and Flickr Faces datasets, we resized the images to be 128x128 to save space. We moved images into three folders (train, val, and test). Each folder had three subdirectories (correct/incorrect/no_mask). This allowed them to be easily loaded with PyTorch's torchvision. If you look at the code snippet, you'll see that we used torchvision's built-in arguments to apply transformations to our images.